// Copyright 2017 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "misc.h"

#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <unistd.h>

#include <unittest/unittest.h>

#include "filesystems.h"

bool fcheck_dir_contents(DIR* dir, expected_dirent_t* edirents, size_t len) {
  BEGIN_HELPER;

  rewinddir(dir);
  size_t seen = 0;
  while (seen != len) {
    struct dirent* de = readdir(dir);
    ASSERT_NE(de, NULL, "Didn't see all expected direntries");
    bool found = false;
    for (size_t i = 0; i < len; i++) {
      if (strcmp(edirents[i].d_name, de->d_name) == 0) {
        ASSERT_EQ(edirents[i].d_type, de->d_type, "Saw direntry with unexpected type");
        ASSERT_FALSE(edirents[i].seen, "Direntry seen twice");
        edirents[i].seen = true;
        seen++;
        found = true;
        break;
      }
    }

    ASSERT_TRUE(found, "Saw an unexpected dirent");
  }

  ASSERT_EQ(readdir(dir), NULL, "There exists an entry we didn't expect to see");

  // Flip 'seen' back to false so the array of expected dirents can be reused
  for (size_t i = 0; i < len; i++) {
    edirents[i].seen = false;
  }

  END_HELPER;
}

bool check_dir_contents(const char* dirname, expected_dirent_t* edirents, size_t len) {
  BEGIN_HELPER;
  DIR* dir = opendir(dirname);
  EXPECT_TRUE(fcheck_dir_contents(dir, edirents, len), "");
  ASSERT_EQ(closedir(dir), 0, "Couldn't close inspected directory");
  END_HELPER;
}

// Check the contents of a file are what we expect
bool check_file_contents(int fd, const uint8_t* buf, size_t length) {
  BEGIN_HELPER;
  ASSERT_EQ(lseek(fd, 0, SEEK_SET), 0, "");
  uint8_t* out = malloc(length);
  ASSERT_NE(out, NULL, "Failed to allocate checking buffer");
  ASSERT_STREAM_ALL(read, fd, out, length);
  ASSERT_EQ(memcmp(buf, out, length), 0, "");
  free(out);
  END_HELPER;
}

bool check_remount(void) {
  BEGIN_HELPER;
  ASSERT_EQ(test_info->unmount(kMountPath), 0, "");
  ASSERT_EQ(test_info->fsck(test_disk_path), 0, "");
  ASSERT_EQ(test_info->mount(test_disk_path, kMountPath), 0, "");
  END_HELPER;
}
